package top.ajunnihao.core.task.tasks;

import org.apache.commons.io.FileUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.context.ServletContextAware;
import top.ajunnihao.core.task.Task;

import javax.servlet.ServletContext;
import java.io.File;

/**
 * TODO 临时文件清理任务
 *
 * @author AJun
 * @version 1.0.0
 * @date 2020/5/31
 */
@Component
public class TmpFileClearTask implements Task, ServletContextAware {

    /**
     * 记录日志
     */
    private static final Logger LOGGER = LoggerFactory.getLogger(TmpFileClearTask.class);

    /**
     * 过期时间: 3小时以上的临时文件
     */
    private final static int EXPIRED = 1000 * 3600 * 3;

    /**
     * 临时文件清理任务需要获取 tmp 目录
     */
    private ServletContext servletContext;

    @Override
    public void setServletContext(ServletContext servletContext) {
        this.servletContext = servletContext;
    }

    @Override
    public void execute() {
        // 临时文件目录
        File tmpDir = new File(servletContext.getRealPath("/tmp"));
        if (!tmpDir.exists()) return;
        File[] files = tmpDir.listFiles();
        if (files == null || files.length == 0) return;

        // 清理过期的文件
        long now = System.currentTimeMillis();
        for (File file : files) {
            // 如果临时文件存在的时间超过三个小时
            if (now - file.lastModified() > EXPIRED) {
                try {
                    LOGGER.info("** 清理过期文件 **: {}", file.getAbsolutePath());
                    // 删除文件
                    FileUtils.deleteQuietly(file);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
    }

}
